from app import db


class ChooseClass(db.Model):
    __tablename__ = 'tbl_chooseclass'

    Stu_id = db.Column(db.String(10), primary_key=True)
    Cos_id = db.Column(db.String(10), primary_key=True)

    def to_json(self):
        return {
            "Stu_id": self.Stu_id,
            "Cos_id": self.Cos_id
        }
